'''
    作者：刘妍
    版本：5.0
    日期：22/05/2020
    1.0功能：计算空气质量指数AQI
    2.0新增功能：读取已经获取的JSON数据文件，并将AQI前5的数据输出到文件
    3.0新增功能：读取已经获取的JSON数据文件，并将其转换为CSV文件
    4.0新增功能：自动识别文件类型（JSON文件还是CSV文件）
    5.0新增功能：利用网络爬虫实时获取城市的空气质量
'''

import requests


def main():
    city_pinyin = input('请输入城市的拼音：')
    # 字符串之间用+，含义为字符串连接
    url = 'http://pm25.in/' + city_pinyin
    # timeout设置了访问连接时间
    r = requests.get(url, timeout=30)
    # print('访问状态码是：{}'.format(r.status_code))
    # print(r.text)

    # 多行字符串用三引号括起来
    aqi_div = '''<div class="span12 data">
        <div class="span1">
          <div class="value">
            '''

    # url_text是网络链接地址url访问获取的当前网页内容all
    url_text = r.text
    index = url_text.find(aqi_div)
    begin_index = index + len(aqi_div)
    end_index = begin_index + 2
    aqi_val = url_text[begin_index: end_index]
    print('空气质量指数为：{}'.format(aqi_val))


if __name__ == '__main__':
    main()